import streamlit as st
import pandas as pd

# 设置页面标题
st.set_page_config(page_title="公司信息查询系统", layout="wide")
st.title("公司信息查询系统")

# 读取Excel文件
@st.cache_data
def load_data():
    df = pd.read_excel("output.xlsx")
    return df

# 加载数据
df = load_data()

# 创建搜索框
search_term = st.text_input("请输入公司简称进行搜索：")

# 搜索功能
if search_term:
    # 使用公司简称进行模糊匹配
    result = df[df["公司简称"].str.contains(search_term, case=False, na=False)]
    
    if len(result) > 0:
        st.success(f"找到 {len(result)} 条匹配记录")
        # 显示查询结果
        st.dataframe(result, use_container_width=True)
    else:
        st.warning("未找到匹配的记录")

# 显示数据统计信息
st.subheader("数据统计")
st.write(f"总记录数：{len(df)}") 